/*
 *Copyright 2021-2023 NERCIS
 *
 *Licensed under the Apache License, Version 2.0 (the "License");
 *you may not use this file except in compliance with the License.
 *You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 *Unless required by applicable law or agreed to in writing, software
 *distributed under the License is distributed on an "AS IS" BASIS,
 *WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *See the License for the specific language governing permissions and
 *limitations under the License.
 */

package cn.ac.nercis.pes.service.project.imported;

import cn.ac.nercis.pes.model.dto.common.Result;
import cn.ac.nercis.pes.model.dto.project.ImportConflictDTO;
import cn.ac.nercis.pes.model.standard.prepare.EvaluationToolSDTO;
import cn.ac.nercis.pes.repository.dal.evaluation.EvaluationToolsRepository;
import cn.ac.nercis.pes.repository.model.evaluation.EvaluationToolsDO;
import cn.ac.nercis.pes.service.utils.CheckFieldUtils;
import lombok.NonNull;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections4.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.time.LocalDateTime;
import java.util.Collections;
import java.util.List;
import java.util.Objects;
import java.util.Optional;
import java.util.stream.Collectors;

/**
 * 测评工具信息导入服务
 *
 * @author zpy
 * @version 1.0.0
 */
@Slf4j
@Service
public class EvaluationToolsImportService {
    private final EvaluationToolsRepository evaluationToolsRepository;

    @Autowired
    public EvaluationToolsImportService(EvaluationToolsRepository evaluationToolsRepository) {
        this.evaluationToolsRepository = evaluationToolsRepository;
    }

    public Result<List<ImportConflictDTO>> checkEvaluationTools(@NonNull String systemId, List<EvaluationToolSDTO> tools) {
        List<EvaluationToolsDO> sourceData = this.evaluationToolsRepository.findAllBySystemId(systemId);
        if (CollectionUtils.isEmpty(sourceData)) {
            return Result.success();
        }
        List<ImportConflictDTO> conflictData = tools.stream()
                .flatMap(ec -> this.checkEvaluationTools(ec, sourceData).stream())
                .filter(Objects::nonNull)
                .collect(Collectors.toList());
        return Result.success(conflictData);
    }

    public Result<String> importEvaluationTools(@NonNull String systemId, List<EvaluationToolSDTO> tools, Boolean cover) {
        List<EvaluationToolsDO> sourceData = this.evaluationToolsRepository.findAllBySystemId(systemId);
        Result<String> result = Result.success();
        for (EvaluationToolSDTO item : tools) {
            result = this.importEvaluationTools(systemId, item, sourceData, cover);
            if (!result.isSuccess()) {
                break;
            }
        }
        return result;
    }

    private List<ImportConflictDTO> checkEvaluationTools(@NonNull EvaluationToolSDTO importData, List<EvaluationToolsDO> sourceData) {
        EvaluationToolsDO entity = sourceData.stream()
                .filter(st -> StringUtils.equals(st.getName(), importData.getName()))
                .findFirst().orElse(null);
        if (Objects.isNull(entity)) {
            return Collections.emptyList();
        }
        return CheckFieldUtils.checkField(entity, importData);
    }

    private Result<String> importEvaluationTools(String systemId, @NonNull EvaluationToolSDTO importData, List<EvaluationToolsDO> sourceData, Boolean cover) {
        EvaluationToolsDO entity = sourceData.stream()
                .filter(st -> StringUtils.equals(st.getName(), importData.getName()))
                .findFirst().orElse(null);
        if (Objects.isNull(entity)) {
            return createEvaluationTools(systemId, importData,sourceData);
        } else {
            return modifyEvaluationTools(entity, importData, cover);
        }
    }

    private Result<String> createEvaluationTools(String systemId, EvaluationToolSDTO importData, List<EvaluationToolsDO> sourceData) {
        EvaluationToolsDO entity = EvaluationToolsDO.builder()
                .systemId(systemId)
                .toolId(importData.getToolId())
                .name(importData.getName())
                .version(importData.getVersion())
                .vendor(importData.getVendor())
                .desc(importData.getDesc())
                .isCommon(importData.getIsCommon())
                .sort(importData.getSort())
                .build();
        entity.setCreateTime(LocalDateTime.now());
        return Optional.of(entity)
                .map(this.evaluationToolsRepository::save)
                .map(data->{
                    sourceData.add(data);
                    return Result.success(data.getId());
                })
                .orElse(Result.failed(String.format("导入%s测评工具信息，创建失败", importData.getName())));
    }

    private Result<String> modifyEvaluationTools(EvaluationToolsDO sourceData, EvaluationToolSDTO importData, Boolean cover) {
        sourceData.setVersion(CheckFieldUtils.applyContent(sourceData.getVersion(), importData.getVersion(), cover));
        sourceData.setVendor(CheckFieldUtils.applyContent(sourceData.getVendor(), importData.getVendor(), cover));
        sourceData.setDesc(CheckFieldUtils.applyContent(sourceData.getDesc(), importData.getDesc(), cover));
        sourceData.setSort(CheckFieldUtils.applyContent(sourceData.getSort(), importData.getSort(), cover));
        sourceData.setModifyTime(LocalDateTime.now());
        return Optional.of(sourceData)
                .map(this.evaluationToolsRepository::save)
                .map(EvaluationToolsDO::getId)
                .map(Result::success)
                .orElse(Result.failed(String.format("导入%s测评工具信息，修改失败", importData.getName())));
    }
}
